import numpy as np
#  TODO： AAMI类别

# 定义分类映射
# 定义分类映射
# - N(正常或者束支传导阻滞节拍)
# - s(室上性异常节拍)
# - V(心室异常节拍)
# - F(融合节拍)
# - Q(未能分类的节拍)
AAMI_MIT = {'N': 'NLRje',  # 将19类信号分为五大类
            'S': 'SAJa',
            'V': 'VE',
            'F': 'F',
            'Q': '/Qf'}
# 5分类存储字典
AAMI = {'N': [], 'S': [], 'V': [], 'F': [], 'Q': []}

ECG = {'N': {}, 'f': {}, 'e': {}, '/': {}, 'j': {}, 'n': {}, 'B': {},
       'L': {}, 'R': {}, 'S': {}, 'A': {}, 'J': {}, 'a': {}, 'V': {},
       'E': {}, 'r': {}, 'F': {}, 'Q': {}, '?': {}}

ECGSignal = {'N': [], 'f': [], 'e': [], '/': [], 'j': [], 'n': [], 'B': [],
             'L': [], 'R': [], 'S': [], 'A': [], 'J': [], 'a': [], 'V': [],
             'E': [], 'r': [], 'F': [], 'Q': [], '?': []}

# 删除非R点的标签
ECG_R_list = np.array(['N', 'f', 'e', '/', 'j', 'n', 'B',
                       'L', 'R', 'S', 'A', 'J', 'a', 'V',
                       'E', 'r', 'F', 'Q', '?'])
